package com.itheima.mp.domain.dto;

import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.lang.func.Func;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.itheima.mp.domain.po.User;
import com.itheima.mp.domain.vo.UserVO;
import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;

import java.util.Collections;
import java.util.List;
import java.util.function.Function;
import java.util.stream.Collectors;

@Data
@ApiModel(description = "分页结果")
public class PageDTO<T> {
    @ApiModelProperty("总条数")
    private Long total;
    @ApiModelProperty("总页数")
    private Long pages;
    @ApiModelProperty("集合")
    private List<T> list;

    public static <PO,VO> PageDTO<VO> of(Page<PO> p, Function<PO,VO> convertor){
        PageDTO<VO> dto=new PageDTO<>();
        //1、总条数
        dto.setTotal(p.getTotal());
        //2、总页数
        dto.setPages(p.getSize());
        //3、当前页数据
        List<PO> records=p.getRecords();
        if(CollUtil.isEmpty(records)){
            dto.setList(Collections.emptyList());
            return dto;
        }
        //4、拷贝user的VO

        List<VO> vos= records.stream().map(convertor).collect(Collectors.toList());
        dto.setList(vos);
        //5返回
        return dto;
    }

//    public static <PO,VO> PageDTO<VO> of(Page<PO> p,Class<VO> clazz){
//        PageDTO<VO> dto=new PageDTO<>();
//        //1、总条数
//        dto.setTotal(p.getTotal());
//        //2、总页数
//        dto.setPages(p.getSize());
//        //3、当前页数据
//        List<PO> records=p.getRecords();
//        if(CollUtil.isEmpty(records)){
//            dto.setList(Collections.emptyList());
//            return dto;
//        }
//        //4、拷贝user的VO
//        List<VO> vos= BeanUtil.copyToList(records,clazz);
//        dto.setList(vos);
//        //5返回
//        return dto;
//    }
}
